package com.ttbj.core.util;

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.nio.channels.FileChannel;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.zip.GZIPOutputStream;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.codec.binary.Hex;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;

/**
 * 文件读写处理工具类
 * 
 * @author Daniel
 * @date 2015年11月3日 下午6:50:48
 * 
 */
@SuppressWarnings({ "rawtypes", "unchecked" })
public class FileUtil {

    private static Logger logger = LoggerFactory.getLogger(FileUtil.class);
    public static final String LOG_HEADER = "文件工具类FileUtil ";
    public static final String DEFAULT_CHARSET_UTF_8 = "UTF-8";

    /** 提供信息摘要算法（MD5 或 SHA 算法），用于生成散列码 */
    static MessageDigest messageDigest = null;
    static {
        try {
            /** 通过getInstance系列静态函数来进行实例化和初始化 */
            messageDigest = MessageDigest.getInstance("MD5");
        } catch (NoSuchAlgorithmException ne) {
            ne.printStackTrace();
            logger.error("{}MessageDigest-异常{}", LOG_HEADER, ne);
        }
    }

    /**
     * 生成文件的MD5值
     * 
     * @author Daniel
     * @date 2016年3月23日 上午11:19:46
     * 
     * @param file
     * @return
     */
    public static String getFileMD5(File file) {
        logger.debug("{}getFileMD5-S file={}", LOG_HEADER, file);
        FileInputStream fileInputStream = null;
        try {
            fileInputStream = new FileInputStream(file);
            byte[] buffer = new byte[8192];
            int length;
            while ((length = fileInputStream.read(buffer)) != -1) {
                /** 通过使用 update 方法处理数据 */
                messageDigest.update(buffer, 0, length);
            }
            /**
             * 调用 digest 方法之一完成哈希计算并返回结果
             * <p>
             * 对于给定数量的更新数据，digest 方法只能被调用一次。digest 方法被调用后，MessageDigest
             * 对象被重新设置成其初始状态。
             */
            return new String(Hex.encodeHex(messageDigest.digest()));
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            logger.error("{}getFileMD5-异常{}", LOG_HEADER, e);
            return null;
        } catch (IOException e) {
            e.printStackTrace();
            logger.error("{}getFileMD5-异常{}", LOG_HEADER, e);
            return null;
        } finally {
            try {
                if (fileInputStream != null) {
                    fileInputStream.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
                logger.error("{}getFileMD5-异常,文件流关闭出错{}", LOG_HEADER, e);
            }
            logger.debug("{}getFileMD5-E", LOG_HEADER);
        }
    }

    /**
     * 上传文件
     * 
     * @author Daniel
     * @date 2016年3月10日 下午3:51:30
     * 
     * @param httpServletRequest 文件上传的请求request
     * @param uploadPath 文件的上传路径
     * @param fileExtension 文件的扩展名
     * @return 返回File，当上传成功时，返回java.io.File
     * @throws IOException
     */
    public static File uploadFile(HttpServletRequest httpServletRequest, String uploadPath, String fileExtension)
            throws IOException {
        logger.debug("{}上传文件-S httpServletRequest={};uploadPath={};fileExtension={}", LOG_HEADER, httpServletRequest,
                uploadPath, fileExtension);
        /** 构建文件保存的目录 **/
        MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) httpServletRequest;
        MultipartFile multipartFile = multipartRequest.getFile("file");
        /** 页面控件的文件流 **/
        if (multipartFile == null) {
        	logger.debug("{}上传文件-E File is not Exsit!",LOG_HEADER);
            return null;
        }
        /** 获取文件的后缀 **/
        String extension = multipartFile.getOriginalFilename().substring(
                multipartFile.getOriginalFilename().lastIndexOf("."));
        if (!StringUtil.isEmpty(fileExtension) && !fileExtension.contains(extension.toUpperCase())) {
        	logger.debug("{}上传文件-E File's Type is error.",LOG_HEADER);
            return null;
        }

        /** 根据路径创建目录 **/
        File fileUploadPath = new File(uploadPath);
        if (!fileUploadPath.exists()) {
            fileUploadPath.mkdirs();
        }
        String fileName = multipartFile.getOriginalFilename();
        File file = new File(uploadPath + File.separator + fileName);
        multipartFile.transferTo(file);
        logger.debug("{}上传文件-E {}", LOG_HEADER, JsonUtil.beanToJson(file));
        return file;
    }

    /**
     * 上传文件
     * 
     * @author Daniel
     * @date 2016年3月10日 下午3:51:30
     * 
     * @param httpServletRequest 文件上传的请求request
     * @param uploadPath 文件的上传路径
     * @return  返回File，当上传成功时，返回java.io.File
     * @throws IOException
     */
    public static File uploadFile(HttpServletRequest httpServletRequest, String uploadPath) throws IOException {
        return uploadFile(httpServletRequest, uploadPath, null);
    }

    /**
     * 读取文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:50:15
     * 
     * @param file 文件全路径
     * @param charset 字符编码
     * @return
     * @throws Exception
     */
    public static String readFile(File file, String charset) throws Exception {
        logger.debug("{}读取文件-S file={};charset={}", LOG_HEADER, file, charset);
        if (file == null) {
            return null;
        }
        if (!file.exists()) {
            throw new Exception("file not exists - " + file.getPath());
        }
        if (charset == null) {
            charset = DEFAULT_CHARSET_UTF_8;
        }

        InputStreamReader in = new InputStreamReader(new FileInputStream(file), charset);
        BufferedReader reader = new BufferedReader(in);
        StringBuffer sb = new StringBuffer();
        String str = reader.readLine();

        while (str != null) {
            sb.append(str);
            sb.append("\n");
            str = reader.readLine();
        }
        /** 删除最后一个多余的\n */
        sb.deleteCharAt(sb.length() - 1);
        reader.close();
        logger.debug("{}读取文件-E {}", LOG_HEADER, sb.toString());
        return sb.toString();
    }

    /**
     * 读取文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:51:17
     * 
     * @param path 文件全路径
     * @param charset 字符编码
     * @return
     * @throws Exception
     */
    public static String readFile(String path, String charset) throws Exception {
        return readFile(new File(path), charset);
    }

    /**
     * 读取文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:51:38
     * 
     * @param path 文件全路径
     * @return
     * @throws Exception
     */
    public static String readFile(String path) throws Exception {
        return readFile(new File(path), null);
    }

    /**
     * 读取文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:51:53
     * 
     * @param file 文件
     * @return
     * @throws Exception
     */
    public static String readFile(File file) throws Exception {
        return readFile(file, null);
    }

    /**
     * 写文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:46:17
     * 
     * @param src 写入字符串
     * @param tarFile 文件全路径
     * @param charset 字符编码
     * @param append true-写入文件末尾处，false-写入文件开始处
     * @throws Exception
     */
    public static void writeFile(String src, File tarFile, String charset, boolean append) throws Exception {
        logger.debug("{}写文件-S src={};tarFile={};charset={};append={}", LOG_HEADER, tarFile, charset, append);
        if (src == null || tarFile == null) {
            return;
        }
        if (!tarFile.getParentFile().exists()) {
            tarFile.getParentFile().mkdirs();
        }
        if (charset == null) {
            charset = DEFAULT_CHARSET_UTF_8;
        }

        OutputStreamWriter out = new OutputStreamWriter(new FileOutputStream(tarFile, append), charset);
        BufferedWriter writer = new BufferedWriter(out);

        writer.write(src);
        writer.close();
        logger.debug("{}写文件-E", LOG_HEADER);
    }

    /**
     * 写文件
     * 
     * @param src
     * @param path
     * @param charset
     * @throws Exception
     */
    /**
     * @author Daniel
     * @date 2015年11月3日 下午6:52:10
     * 
     * @param src 写入字符串
     * @param path 文件全路径
     * @param charset 字符编码
     * @param append true-写入文件末尾处，false-写入文件开始处
     * @throws Exception
     */
    public static void writeFile(String src, String path, String charset, boolean append) throws Exception {
        writeFile(src, new File(path), charset, append);
    }

    /**
     * 写文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:53:18
     * 
     * @param src 写入字符串
     * @param path 文件全路径
     * @param append true-写入文件末尾处，false-写入文件开始处
     * 
     * @throws Exception
     */
    public static void writeFile(String src, String path, boolean append) throws Exception {
        writeFile(src, new File(path), null, append);
    }

    /**
     * 写文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:54:14
     * 
     * @param src 写入字符串
     * @param file 文件
     * @param append true-写入文件末尾处，false-写入文件开始处
     * @throws Exception
     */
    public static void writeFile(String src, File file, boolean append) throws Exception {
        writeFile(src, file, null, append);
    }

    /**
     * 写文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:54:46
     * 
     * @param src 写入字符串
     * @param tarFile 目标文件
     * @param charset 字符编码
     * @throws Exception
     */
    public static void writeFile(String src, File tarFile, String charset) throws Exception {
        writeFile(src, tarFile, charset, false);
    }

    /**
     * 写文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:55:19
     * 
     * @param src 写入字符串
     * @param path 文件全路径
     * @param charset 字符编码
     * @throws Exception
     */
    public static void writeFile(String src, String path, String charset) throws Exception {
        writeFile(src, new File(path), charset);
    }

    /**
     * 写文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:55:32
     * 
     * @param src 写入字符串
     * @param path 文件全路径
     * @throws Exception
     */
    public static void writeFile(String src, String path) throws Exception {
        writeFile(src, new File(path), null);
    }

    /**
     * 写文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:55:54
     * 
     * @param src 写入字符串
     * @param file 文件
     * @throws Exception
     */
    public static void writeFile(String src, File file) throws Exception {
        writeFile(src, file, null);
    }

    /**
     * 创建Gz压缩文件
     * 
     * @author Daniel
     * @date 2015年11月3日 下午6:45:24
     * 
     * @param fileName 创建Gz的文件名
     * @throws Exception
     */
    public static void createGZ(String fileName) throws Exception {
        logger.debug("{}创建Gz压缩文件-S src={};fileName={}", LOG_HEADER, fileName);
        String outFilename = fileName + ".gz";
        GZIPOutputStream out = new GZIPOutputStream(new FileOutputStream(outFilename));
        String inFilename = fileName;
        FileInputStream in = new FileInputStream(inFilename);
        byte[] buf = new byte[1024];
        int len;
        while ((len = in.read(buf)) > 0) {
            out.write(buf, 0, len);
        }
        in.close();
        out.finish();
        out.close();
        logger.debug("{}创建Gz压缩文件-E");
    }

    /**
     * 获取文件的大小
     * 
     * @author Daniel
     * @date 2016年3月23日 上午11:42:01
     * 
     * @param file
     * @return long 大小
     */
    public static long getFileSize(File file) {
        logger.debug("{}获取文件的大小-S file={}", LOG_HEADER, file);
        FileChannel fc = null;
        try {
            if (file.exists() && file.isFile()) {
                FileInputStream fis = new FileInputStream(file);
                fc = fis.getChannel();
                long size = fc.size();
                logger.info("{}获取文件的大小-E size={};", size);
                return size;
            } else {
                logger.info("{}获取文件的大小-E ; file doesn't exist or is not a file");
                return 0L;
            }
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            logger.error("{}获取文件的大小-异常{}", LOG_HEADER, e);
            return 0L;
        } catch (IOException e) {
            e.printStackTrace();
            logger.error("{}获取文件的大小-异常{}", LOG_HEADER, e);
            return 0L;
        } finally {
            if (null != fc) {
                try {
                    fc.close();
                } catch (IOException e) {
                    e.printStackTrace();
                    logger.error("{}获取文件的大小-异常{}", LOG_HEADER, e);
                }
            }
        }
    }
}
